from selenium import webdriver
from selenium.webdriver.common.by import By
import time

# 启动 Edge 浏览器
driver = webdriver.Edge()
print("Edge 浏览器已启动。")

# 打开页面
driver.get("file:///F:/TestProject/register.html")
print(f"页面已打开：{driver.current_url}")

# 切换到注册用户页面的 iframe
try:
    driver.switch_to.frame("myframe1")  # 通过 name 切换
    print("切换到注册用户页面的 iframe。")
except Exception as e:
    print(f"切换到注册用户页面的 iframe 时出错：{e}")

# 操作注册用户页面中的元素
try:
    driver.find_element(By.ID, "AuserA").send_keys("test_user")  # 输入用户名
    print("输入用户名：test_user")
    driver.find_element(By.ID, "passwordA").send_keys("test_password")  # 输入密码
    print("输入密码：test_password")
    driver.find_element(By.ID, "telA").send_keys("1234567890")  # 输入电话号码
    print("输入电话号码：1234567890")
    driver.find_element(By.ID, "emailA").send_keys("test@example.com")  # 输入电子邮箱
    print("输入电子邮箱：test@example.com")
except Exception as e:
    print(f"操作注册用户页面元素时出错：{e}")

# 截图
driver.save_screenshot("register_page.png")
print("截图已保存为 'register_page.png'。")

# 切换回主页面
driver.switch_to.default_content()
print("切换回主页面。")

# 切换到登录用户页面的 iframe
try:
    driver.switch_to.frame("myframe2")  # 通过 name 切换
    print("切换到登录用户页面的 iframe。")
except Exception as e:
    print(f"切换到登录用户页面的 iframe 时出错：{e}")

# 操作登录用户页面中的元素
try:
    driver.find_element(By.ID, "BuserA").send_keys("test_user")  # 输入用户名
    print("输入用户名：test_user")
    driver.find_element(By.ID, "passwordA").send_keys("test_password")  # 输入密码
    print("输入密码：test_password")
    driver.find_element(By.TAG_NAME, "button").click()  # 点击登录按钮
    print("点击登录按钮。")
except Exception as e:
    print(f"操作登录用户页面元素时出错：{e}")

# 截图
driver.save_screenshot("login_page.png")
print("截图已保存为 'login_page.png'。")

# 切换回主页面
driver.switch_to.default_content()
print("切换回主页面。")

# 关闭浏览器
driver.quit()
print("浏览器已关闭。")